package cn.itcast.web.shiro;
import cn.itcast.domain.system.Module;
import cn.itcast.domain.system.User;
import cn.itcast.service.system.ModuleService;
import cn.itcast.service.system.UserService;
import org.apache.shiro.authc.*;
import org.apache.shiro.authz.AuthorizationInfo;
import org.apache.shiro.authz.SimpleAuthorizationInfo;
import org.apache.shiro.realm.AuthorizingRealm;
import org.apache.shiro.subject.PrincipalCollection;
import org.springframework.beans.factory.annotation.Autowired;

import java.util.List;

/**
 * 自定义的realm类，提供认证、授权数据访问的方法
 */
public class AuthRealm extends AuthorizingRealm{

    @Autowired
    private UserService userService;
    @Autowired
    private ModuleService moduleService;

    // 认证
    @Override
    protected AuthenticationInfo doGetAuthenticationInfo(AuthenticationToken token) throws AuthenticationException {

        // 转换
        UsernamePasswordToken upToken = (UsernamePasswordToken) token;
        // 获取登陆名
        String email = upToken.getUsername();
        // 根据email查询
        User user = userService.findByEmail(email);
        if (user == null) {
            return null; // UnknownAccountException
        }
        // 获取数据库中正确的密码
        String dbPwd = user.getPassword();

        // 参数1：身份对象（从数据库中查询到的user对象）
        // 参数2：数据库中正确的密码
        // 参数3：realm的名称; 可以随便写，但是有多个realm要唯一。getName()获取默认名称
        SimpleAuthenticationInfo sai =
                new SimpleAuthenticationInfo(user,dbPwd,getName());
        return sai;
    }

    // 授权
    @Override
    protected AuthorizationInfo doGetAuthorizationInfo(PrincipalCollection principals) {
        System.out.println("将用户拥有的权限写入shiro缓存!!");
        /* 查询登陆用户的权限*/
        //1. 获取用户的身份对象（认证方法返回对象的构造函数的第一个参数）
        User user = (User) principals.getPrimaryPrincipal();

        //2. 根据用户的id查询权限
        List<Module> moduleList = moduleService.findModuleByUserId(user.getId());

        //3. 返回，告诉shiro当前登陆用户拥有的权限。后面就可以进行权限校验（四种方式）
        SimpleAuthorizationInfo sai = new SimpleAuthorizationInfo();
        if (moduleList != null && moduleList.size() > 0){
            for (Module module : moduleList) {
                // 返回用户的权限
                sai.addStringPermission(module.getName());
            }
        }
        return sai;
    }

}
